optimize Python code

welcome to shbcf.ru